/*借助条件变量模拟 生产者-消费者 问题*/
#include <stdlib.h>
#include <unistd.h>
#include <pthread.h>
#include <stdio.h>

void err_thread(int ret, char *str)
{
    if (ret != 0)
    {
        fprintf(stderr, "%s:%s\n", str, strerror(ret));
        pthread_exit(NULL);
    }
}

/*链表作为公享数据,需被互斥量保护*/
struct msg 
{
    struct msg *next;
    int num;
};

struct msg *head;// 链表头指针

/* 静态初始化 一个条件变量 和 一个互斥量*/
pthread_cond_t has_product = PTHREAD_COND_INITIALIZER;   // 定义初始化一个条件变量
pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;        // 定义初始化一个互斥量

void *consumer(void *p)// 消费者
{
    struct msg *mp;

    while(1) 
    {
        pthread_mutex_lock(&lock);// 加锁 互斥量
        // 判断
        while (head == NULL) 
        { 
            //头指针为空,说明没有节点 可以为 if 吗
            pthread_cond_wait(&has_product, &lock); // 阻塞等待条件变量
                                                    // pthread_cond_wait 返回时，重写加锁
        }
        mp = head;
        head = mp->next; //模拟消费掉一个产品
        // 解锁
        pthread_mutex_unlock(&lock);

        printf("-Consume %lu---%d\n", pthread_self(), mp->num);
        free(mp);// 原因是: 共享
        sleep(rand() % 5);
    }

    return NULL;
}

void *producer(void *p)// 生产者
{
    struct msg *mp;

    while(1) 
    {
        mp = malloc(sizeof(struct msg));
        mp->num = rand() % 1000 + 1; //模拟生产一个产品
        printf("-Produce ---------------------%d\n", mp->num);
        pthread_mutex_lock(&lock);// 加锁
        // 头插法
        // 写公共区域
        mp->next = head;
        head = mp;
        pthread_mutex_unlock(&lock);

        // 通知
        pthread_cond_signal(&has_product); //将等待在该条件变量has_product 上的一个线程唤醒
        sleep(rand() % 5);
    }

    return NULL;
}

int main(int argc, char *argv[])
{
    int ret;
    pthread_t pid, cid;
    srand(time(NULL));// 随机数种子

    ret = pthread_create(&pid, NULL, producer, NULL);// 生产者
    if (ret != 0)
    {
        err_thread(ret, "pthread_create producer error");
    }

    ret = pthread_create(&cid, NULL, consumer, NULL);// 消费者
    if (ret != 0)
    {
        err_thread(ret, "pthread_create consumer error");
    }

    pthread_join(pid, NULL);
    pthread_join(cid, NULL);

    return 0;
}